Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程) |
您所在的位置:网站首页 › pytorch 调用 海光gpu › Anaconda环境中PyTorch GPU版本安装(史上最全面,适用于新手的教程) |
一.准备anaconda
没有anaconda可以去官网下载一下,https://www.anaconda.com/,有人觉得官网下载过慢的话,这里给你提供一个镜像网址:https://mirrors.bfsu.edu.cn/anaconda/archive/ 安装步骤: 1.在开始程序中打开Anaconda Prompt 2.在步骤3之前,为了防止anaconda安装pytorch过慢或者失败,这里先设置清华源,里面的命令要一行一行执行 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/ conda config --set show_channel_urls yes这里所输入的命令是没有任何提示的 3.通过conda create -n pytorch python=3.9命令创建pytorch环境,python的版本可以自由选择 等待下载 我们要选择自己电脑对应的cuda版本,可以通过进入电脑的命令界面(window+R,然后输入cmd),使用nvidia-smi命令查看,有些人肯定会显示如下图的错误: 因为它找不到该命令。这个文件是一个exe文件,一般都在下面这个文件夹中。 C:\Program Files\NVIDIA Corporation\NVSMI 解决方案1.添加环境变量:在path里面添加上面的路径,就可以访问到这个命令,就可以执行了。 2.更改命令路径:使用cd命令修改cmd中的文件夹,跳转到 ‘nvidia-smi’ 命令所在的文件夹。然后再次输入该命令即可。 cd C:\Program Files\NVIDIA Corporation\NVSMI nvidia-smi执行之后可以看到如下信息: 解决:进入这个网址https://www.nvidia.com/download/index.aspx?lang=cn选择我们自己显卡对应的型号,然后下载进行升级。 解决: 1.通过下载鲁大师等软件进行检测 2.通过进入任务管理器,点击性能选项,往下翻就可以看到
在pytorch环境下写入复制的命令后,输入y,然后等待下载安装
|
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |